import dayjs from "dayjs";

/**
 * 验证身份证
 * */
class ValidateIdent
{
    // 城市代码列表，用于校验身份证号码的省份部分
    aIdentityCodeCity: Record<string, string> = {
        11: "北京", 12: "天津", 13: "河北", 14: "山西", 15: "内蒙古", 21: "辽宁", 22: "吉林",
        23: "黑龙江", 31: "上海", 32: "江苏", 33: "浙江", 34: "安徽", 35: "福建", 36: "江西",
        37: "山东", 41: "河南", 42: "湖北", 43: "湖南", 44: "广东", 45: "广西", 46: "海南",
        50: "重庆", 51: "四川", 52: "贵州", 53: "云南", 54: "西藏", 61: "陕西", 62: "甘肃",
        63: "青海", 64: "宁夏", 65: "新疆", 71: "台湾", 81: "香港", 82: "澳门", 91: "国外 "
    };

    // 构造函数
    constructor()
    {
    }

    /**
     * 身份证号码检验主入口
     * @param card 身份证号码
     * @returns 是否身份证号码有效
     */
    validate(card: string): boolean
    {
        // 初始化验证结果为 true
        let pass = true;

        // 检查身份证号码是否为空
        if (pass && card === "")
        {
            pass = false;
        }

        // 将身份证号码转换为大写,因为最后一位可能为X
        card = card.toLocaleUpperCase();

        // 检查身份证号码的格式
        if (pass && !this.IdentityCodeIsCardNo(card))
        {
            pass = false;
        }

        // 检查身份证号码的省份
        if (pass && !this.IdentityCodeCheckProvince(card))
        {
            pass = false;
        }

        // 检查身份证号码的生日
        if (pass && !this.IdentityCodeCheckBirthday(card).validate)
        {
            pass = false;
        }

        // 检查身份证号码的校验位
        if (pass && !this.IdentityCodeCheckParity(card))
        {
            pass = false;
        }

        // 返回最终的验证结果
        return pass;
    }

    /**
     * 通过身份证号码获取生日入口
     * @param card 身份证号码
     * @returns {number} 当前年龄
     */
    getBirthday(card: string): {year: string; month: string; day: string; age: number}
    {
        const pass = this.validate(card);

        if (pass)
        {
            const {year = "", month = "", day = ""} = this.IdentityCodeCheckBirthday(card) || {};

            return {
                year, month, day, age: dayjs().diff(`${year}-${month}-${day}`, "years")
            };
        }
        else
        {
            return {year: "", month: "", day: "", age: -1};
        }
    }

    /**
     * 通过身份证号码获取性别入口
     * @param {string} card 身份证号码
     * @returns {number} 性别，1为男性，0为女性，-1为未知
     * */
    getSex(card: string): number
    {
        const pass = this.validate(card);

        if (pass)
        {
            const idCard = this.IdentityCodeChangeFifteenToEighteen(card);

            return parseInt(idCard.charAt(16)) % 2 === 0 ? 0 : 1;
        }
        else
        {
            return -1;
        }
    }

    /**
     * 验证身份证号码的格式
     * @param card 身份证号码
     * @returns 是否符合格式
     */
    private IdentityCodeIsCardNo(card: string): boolean
    {
        // 定义正则表达式，身份证号码为15位或者18位
        // 15位时全为数字，18位前17位为数字，最后一位是校验位，可能为数字或字符X
        const reg = /(^\d{15}$)|(^\d{17}(\d|X)$)/;

        // 使用正则表达式测试输入的身份证号码
        return reg.test(card);
    }

    /**
     * 校验身份证号码的省份
     * @param card 身份证号码
     * @returns 省份名称，如果省份不存在则返回 undefined
     */
    private IdentityCodeCheckProvince(card: string): string | void
    {
        // 取身份证号码的前两位，表示省份代码
        const province = card.substr(0, 2);

        // 根据省份代码从城市代码列表中查找对应的省份名称
        return this.aIdentityCodeCity[province];
    }

    /**
     * 校验身份证号码的生日
     * @param card 身份证号码
     * @returns 是否生日有效
     */
    private IdentityCodeCheckBirthday(card: string): {year: string; month: string; day: string; validate: boolean}
    {
        // 获取身份证号码的长度
        const len = card.length;

        // 处理15位身份证号码
        if (len === 15)
        {
            // 定义正则表达式，匹配15位身份证号码的结构
            const reFifteen = /^(\d{6})(\d{2})(\d{2})(\d{2})(\d{3})$/;

            // 使用正则表达式提取出生年月日
            const arrData: string[] | null = card.match(reFifteen);

            let year = "", month = "", day = "";

            // 如果匹配成功，提取年月日
            if (arrData && arrData.length)
            {
                year = arrData[2];
                month = arrData[3];
                day = arrData[4];
            }

            // 生成完整的出生日期（15位身份证号码的年份前补19）
            const birthday = new Date("19" + year + "/" + month + "/" + day);

            // 调用验证生日的方法
            return {
                year, month, day, validate: this.IdentityCodeVerifyBirthday(year, month, day, birthday)
            };
        }

        // 处理18位身份证号码
        if (len === 18)
        {
            // 定义正则表达式，匹配18位身份证号码的结构
            const reEighteen = /^(\d{6})(\d{4})(\d{2})(\d{2})(\d{3})([0-9]|X)$/;

            // 使用正则表达式提取出生年月日
            const arrData = card.match(reEighteen);

            let year = "", month = "", day = "";

            // 如果匹配成功，提取年月日
            if (arrData && arrData.length)
            {
                year = arrData[2];
                month = arrData[3];
                day = arrData[4];
            }

            // 生成完整的出生日期
            const birthday = new Date(Number(year), Number(month) - 1, Number(day));

            // 调用验证生日的方法
            return {
                year, month, day, validate: this.IdentityCodeVerifyBirthday(year, month, day, birthday)
            };
        }

        // 如果身份证号码长度不符合要求，返回 false
        return {
            year: "", month: "", day: "", validate: false
        };
    }

    /**
     * 校验生日日期是否合理
     * @param year 年份
     * @param month 月份
     * @param day 日期
     * @param birthday 生成的日期对象
     * @returns 是否日期合理
     */
    private IdentityCodeVerifyBirthday(year: string, month: string, day: string, birthday: Date): boolean
    {
        // 获取当前日期
        const now = new Date();
        const nowYear = now.getFullYear();

        // 将字符串类型的年月日转换为数字类型
        const transformYear = Number(year);
        const transformMonth = Number(month);
        const transformDay = Number(day);

        // 检查生成的日期对象是否与提取的年月日一致
        if (birthday.getFullYear() === transformYear && birthday.getMonth() + 1 === transformMonth && birthday.getDate() === transformDay)
        {
            // 计算年龄
            const time = nowYear - transformYear;

            // 判断年龄是否在0岁到150岁之间
            return time >= 0 && time <= 150;
        }

        // 如果日期不一致，返回 false
        return false;
    }

    /**
     * 校验身份证号码的校验位
     * @param card 身份证号码
     * @returns 是否校验位正确
     */
    private IdentityCodeCheckParity(card: string): boolean
    {
        // 将15位身份证号码转换为18位
        card = this.IdentityCodeChangeFifteenToEighteen(card);

        // 获取身份证号码的长度
        const len = card.length;

        // 处理18位身份证号码
        if (len === 18)
        {
            // 定义权重数组
            const arrInt = [7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2];

            // 定义校验码数组
            const arrCh = ["1", "0", "X", "9", "8", "7", "6", "5", "4", "3", "2"];

            // 初始化累加值
            let cardTemp = 0;

            // 计算前17位数字与权重的乘积之和
            for (let i = 0; i < 17; i++)
            {
                cardTemp += Number(card.substr(i, 1)) * arrInt[i];
            }

            // 计算校验码
            const valNum = arrCh[cardTemp % 11];

            // 比较计算的校验码与身份证号码的最后一位
            return valNum === card.substr(17, 1);
        }

        // 如果身份证号码长度不符合要求，返回 false
        return false;
    }

    /**
     * 将15位身份证号码转换为18位
     * @param card 15位身份证号码
     * @returns 18位身份证号码
     */
    private IdentityCodeChangeFifteenToEighteen(card: string): string
    {
        // 获取身份证号码的长度
        if (card.length === 15)
        {
            // 定义权重数组
            const arrInt = [7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2];

            // 定义校验码数组
            const arrCh = ["1", "0", "X", "9", "8", "7", "6", "5", "4", "3", "2"];

            // 初始化累加值
            let cardTemp = 0;

            // 将15位身份证号码的前6位和后9位拼接，并在中间插入19，形成17位数字
            card = card.substr(0, 6) + "19" + card.substr(6, card.length - 6);

            // 计算前17位数字与权重的乘积之和
            for (let i = 0; i < 17; i++)
            {
                cardTemp += Number(card.substr(i, 1)) * arrInt[i];
            }

            // 计算校验码
            const valNum = arrCh[cardTemp % 11];

            // 将校验码添加到17位数字的末尾，形成18位身份证号码
            card += valNum;

            // 返回18位身份证号码
            return card;
        }

        // 如果身份证号码长度不符合要求，返回原号码
        return card;
    }
}

export default ValidateIdent;